import baize_rag
import baize_config

"""URL解析"""
print(baize_rag.get_webpage("https://www.openkylin.top/community/aboutus-cn.html"))

"""音频转录"""
s2t = baize_rag.Speech2Text(baize_config.DEFAULT_WHISPER_MODEL_PATH)
print(s2t.transcribe("./test_sample/平头哥.wav"))

"""PDF解析"""
print(baize_rag.pdf2text("./test_sample/白泽简介.pdf"))
